// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/android/scoped_java_ref.h"

#include "base/android/jni_android.h"
#include "base/logging.h"

namespace base {
namespace android {
    namespace {

        const int kDefaultLocalFrameCapacity = 16;

    } // namespace

    ScopedJavaLocalFrame::ScopedJavaLocalFrame(JNIEnv* env)
        : env_(env)
    {
        int failed = env_->PushLocalFrame(kDefaultLocalFrameCapacity);
        DCHECK(!failed);
    }

    ScopedJavaLocalFrame::ScopedJavaLocalFrame(JNIEnv* env, int capacity)
        : env_(env)
    {
        int failed = env_->PushLocalFrame(capacity);
        DCHECK(!failed);
    }

    ScopedJavaLocalFrame::~ScopedJavaLocalFrame() { env_->PopLocalFrame(NULL); }

#if DCHECK_IS_ON()
    // This constructor is inlined when DCHECKs are disabled; don't add anything
    // else here.
    JavaRef<jobject>::JavaRef(JNIEnv* env, jobject obj)
        : obj_(obj)
    {
        if (obj) {
            DCHECK(env && env->GetObjectRefType(obj) == JNILocalRefType);
        }
    }
#endif

    JNIEnv* JavaRef<jobject>::SetNewLocalRef(JNIEnv* env, jobject obj)
    {
        if (!env) {
            env = AttachCurrentThread();
        } else {
            DCHECK_EQ(env, AttachCurrentThread()); // Is |env| on correct thread.
        }
        if (obj)
            obj = env->NewLocalRef(obj);
        if (obj_)
            env->DeleteLocalRef(obj_);
        obj_ = obj;
        return env;
    }

    void JavaRef<jobject>::SetNewGlobalRef(JNIEnv* env, jobject obj)
    {
        if (!env) {
            env = AttachCurrentThread();
        } else {
            DCHECK_EQ(env, AttachCurrentThread()); // Is |env| on correct thread.
        }
        if (obj)
            obj = env->NewGlobalRef(obj);
        if (obj_)
            env->DeleteGlobalRef(obj_);
        obj_ = obj;
    }

    void JavaRef<jobject>::ResetLocalRef(JNIEnv* env)
    {
        if (obj_) {
            DCHECK_EQ(env, AttachCurrentThread()); // Is |env| on correct thread.
            env->DeleteLocalRef(obj_);
            obj_ = NULL;
        }
    }

    void JavaRef<jobject>::ResetGlobalRef()
    {
        if (obj_) {
            AttachCurrentThread()->DeleteGlobalRef(obj_);
            obj_ = NULL;
        }
    }

    jobject JavaRef<jobject>::ReleaseInternal()
    {
        jobject obj = obj_;
        obj_ = NULL;
        return obj;
    }

} // namespace android
} // namespace base
